//
// Created by xu on 2024/11/22.
//

#include <stdio.h>
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>

#define ERROR_BUF(ret) \
    char errbuf[1024]; \
    av_strerror(ret, errbuf, sizeof (errbuf));

// 输入缓冲区的大小
#define IN_DATA_SIZE 20480
// 需要再次读取输入文件数据的阈值
#define REFILL_THRESH 4096

typedef struct {
    const char *filename;
    int sampleRate;
    enum AVSampleFormat sampleFormat;
    AVChannelLayout chLayout;
} AudioDecodeSpec;

static int decode(AVCodecContext *ctx, AVPacket *pkt, AVFrame *frame, FILE *outFile) {
    // 发送压缩数据到解码器
    int ret = avcodec_send_packet(ctx, pkt);
    if (ret < 0) {
        ERROR_BUF(ret);
        av_log(NULL, AV_LOG_INFO, "avcodec_send_packet error %s\n", errbuf);
        return ret;
    }

    while (1) {
        // 获取解码后的数据
        ret = avcodec_receive_frame(ctx, frame);
        if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
            return 0;
        } else if (ret < 0) {
            ERROR_BUF(ret);
            av_log(NULL, AV_LOG_INFO, "avcodec_receive_frame error %s\n", errbuf);
            return ret;
        }
        // 将解码后的数据写入文件
        fwrite(frame->data[0], 1, frame->linesize[0], outFile);
    }
}

static void aac_decode(const char *inFilename, AudioDecodeSpec *out) {
    int ret = 0;
    int inLen = 0; // 每次从输入文件中读取的长度
    int inEnd = 0; // 是否已经读取到了输入文件的尾部

    // 用来存放读取的文件数据
    // 加上AV_INPUT_BUFFER_PADDING_SIZE是为了防止某些优化过的reader一次性读取过多导致越界
    char inDataArray[IN_DATA_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
    char *inData = inDataArray;

    FILE *inFile = NULL;
    FILE *outFile = NULL;

    // 解码器
    const AVCodec *codec = NULL;
    // 上下文
    AVCodecContext *ctx = NULL;
    // 解析器上下文
    AVCodecParserContext *parserCtx = NULL;
    // 存放解码前的数据
    AVPacket *pkt = NULL;
    // 存放解码后的数据
    AVFrame *frame = NULL;

    // 获取解码器
    codec = avcodec_find_decoder_by_name("libfdk_aac");
    if (!codec) {
        av_log(NULL, AV_LOG_INFO, "decoder libfdk_aac not found\n");
        return;
    }
    // 获取解析器上下文
    parserCtx = av_parser_init(codec->id);
    if (!parserCtx) {
        av_log(NULL, AV_LOG_INFO, "av_parser_init error\n");
        return;
    }

    // 创建上下文
    ctx = avcodec_alloc_context3(codec);
    if (!ctx) {
        av_log(NULL, AV_LOG_INFO, "avcodec_alloc_context3 error\n");
        goto end;
    }

    // 创建AVPacket
    pkt = av_packet_alloc();
    if (!pkt) {
        av_log(NULL, AV_LOG_INFO, "av_packet_alloc error\n");
        goto end;
    }

    // 创建AVFrame
    frame = av_frame_alloc();
    if (!frame) {
        av_log(NULL, AV_LOG_INFO, "av_frame_alloc error\n");
        goto end;
    }

    // 打开解码器
    ret = avcodec_open2(ctx, codec, NULL);
    if (ret < 0) {
        ERROR_BUF(ret);
        av_log(NULL, AV_LOG_INFO, "avcodec_open2 error %s\n", errbuf);
        goto end;
    }

    // 打开文件
    inFile = fopen(inFilename, "rb");
    if (inFile == NULL) {
        av_log(NULL, AV_LOG_INFO, "File open error %s\n", inFilename);
        goto end;
    }
    outFile = fopen(out->filename, "wb");
    if (outFile == NULL) {
        av_log(NULL, AV_LOG_INFO, "File open error %s\n", out->filename);
        goto end;
    }

    // 读取数据
    inLen = fread(inData, 1, IN_DATA_SIZE, inFile);
    while (inLen > 0) {
        // 经过解析器上下文处理
        ret = av_parser_parse2(parserCtx, ctx, &pkt->data, &pkt->size,
                               (uint8_t *) inData, inLen, AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
        if (ret < 0) {
            ERROR_BUF(ret);
            av_log(NULL, AV_LOG_INFO, "av_parser_parse2 error %s\n", errbuf);
            goto end;
        }
        // 跳过已经解析过的数据
        inData += ret;
        // 减去已经解析过的数据大小
        inLen -= ret;

        // 解码
        if (pkt->size > 0 && decode(ctx, pkt, frame, outFile) < 0) {
            goto end;
        }

        // 如果数据不够了，再次读取文件
        if (inLen < REFILL_THRESH && !inEnd) {
            // 剩余的数据移动到缓冲区前
            memmove(inDataArray, inData, inLen);
            inData = inDataArray;

            // 跨过已有数据，读取文件数据
            int len = fread(inData + inLen, 1, IN_DATA_SIZE - inLen, inFile);
            if (len > 0) {
                inLen += len;
            } else {
                inEnd = 1;
            }
        }
    }

    // flush解码器
    decode(ctx, NULL, frame, outFile);
    // 设置输出参数
    out->sampleFormat = ctx->sample_fmt;
    out->sampleRate = ctx->sample_rate;
    out->chLayout = ctx->ch_layout;

    end:
    fclose(inFile);
    fclose(outFile);
    av_frame_free(&frame);
    av_packet_free(&pkt);
    av_parser_close(parserCtx);
    avcodec_free_context(&ctx);
}


int main(int argc, char *argv[]) {
    AudioDecodeSpec out;
    out.filename = "C:/Users/xu/Desktop/output_2.pcm";
    aac_decode("C:/Users/xu/Desktop/output.aac", &out);
    av_log(NULL, AV_LOG_INFO, "采样率： %d\n", out.sampleRate);
    av_log(NULL, AV_LOG_INFO, "采样格式： %s\n", av_get_sample_fmt_name(out.sampleFormat));
    av_log(NULL, AV_LOG_INFO, "声道数： %d\n", out.chLayout.nb_channels);
    return 0;
}